/*
 * Copyright 2019-2025 the original author or authors.
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 * https://www.apache.org/licenses/LICENSE-2.0
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package com.iiifi.kite.core.util;

import java.io.*;
import java.nio.charset.Charset;
import java.nio.file.Files;
import java.nio.file.Path;
import java.nio.file.Paths;
import java.util.ArrayList;
import java.util.List;

import org.springframework.lang.Nullable;
import org.springframework.util.Assert;
import org.springframework.util.FileSystemUtils;
import org.springframework.util.PatternMatchUtils;
import org.springframework.util.StringUtils;
import org.springframework.web.multipart.MultipartFile;

import lombok.experimental.UtilityClass;

/**
 * 文件工具类
 *
 * @author kite@iiifi.com 花朝
 */
@UtilityClass
public class FileUtils extends org.springframework.util.FileCopyUtils {

    /**
     * 默认为true
     */
    public static class TrueFilter implements FileFilter, Serializable {
        private static final long serialVersionUID = -6420452043795072619L;

        public final static TrueFilter TRUE = new TrueFilter();

        @Override
        public boolean accept(File pathname) {
            return true;
        }
    }

    /**
     * 扫描目录下的文件
     *
     * @param path 路径
     * @return 文件集合
     */
    public static List<File> list(String path) {
        File file = new File(path);
        return list(file, TrueFilter.TRUE);
    }

    /**
     * 扫描目录下的文件
     *
     * @param path 路径
     * @param fileNamePattern 文件名 * 号
     * @return 文件集合
     */
    public static List<File> list(String path, final String fileNamePattern) {
        File file = new File(path);
        return list(file, pathname -> {
            String fileName = pathname.getName();
            return PatternMatchUtils.simpleMatch(fileNamePattern, fileName);
        });
    }

    /**
     * 扫描目录下的文件
     *
     * @param path 路径
     * @param filter 文件过滤
     * @return 文件集合
     */
    public static List<File> list(String path, FileFilter filter) {
        File file = new File(path);
        return list(file, filter);
    }

    /**
     * 扫描目录下的文件
     *
     * @param file 文件
     * @return 文件集合
     */
    public static List<File> list(File file) {
        List<File> fileList = new ArrayList<>();
        return list(file, fileList, TrueFilter.TRUE);
    }

    /**
     * 扫描目录下的文件
     *
     * @param file 文件
     * @param fileNamePattern "xxx*", "*xxx", "*xxx*" and "xxx*yyy"
     * @return 文件集合
     */
    public static List<File> list(File file, final String fileNamePattern) {
        List<File> fileList = new ArrayList<>();
        return list(file, fileList, pathname -> {
            String fileName = pathname.getName();
            return PatternMatchUtils.simpleMatch(fileNamePattern, fileName);
        });
    }

    /**
     * 扫描目录下的文件
     *
     * @param file 文件
     * @param fileNamePatterns "xxx*", "*xxx", "*xxx*" and "xxx*yyy"
     * @return 文件集合
     */
    public static List<File> list(File file, final String... fileNamePatterns) {
        List<File> fileList = new ArrayList<>();
        return list(file, fileList, pathname -> {
            String fileName = pathname.getName();
            return PatternMatchUtils.simpleMatch(fileNamePatterns, fileName);
        });
    }

    /**
     * 扫描目录下的文件
     *
     * @param file 文件
     * @param filter 文件过滤
     * @return 文件集合
     */
    public static List<File> list(File file, FileFilter filter) {
        List<File> fileList = new ArrayList<>();
        return list(file, fileList, filter);
    }

    /**
     * 扫描目录下的文件
     *
     * @param file 文件
     * @param filter 文件过滤
     * @return 文件集合
     */
    private static List<File> list(File file, List<File> fileList, FileFilter filter) {
        if (file.isDirectory()) {
            File[] files = file.listFiles();
            if (files != null) {
                for (File f : files) {
                    list(f, fileList, filter);
                }
            }
        } else {
            // 过滤文件
            boolean accept = filter.accept(file);
            if (file.exists() && accept) {
                fileList.add(file);
            }
        }
        return fileList;
    }

    /**
     * 获取文件后缀名
     *
     * @param path 文件路径
     * @return {String}
     */
    @Nullable
    public static String getFilename(@Nullable String path) {
        return org.springframework.util.StringUtils.getFilename(path);
    }

    /**
     * 获取文件后缀名 jpg
     *
     * @param fullName 文件全名
     * @return {String}
     */
    @Nullable
    public static String getFileExtension(@Nullable String fullName) {
        return org.springframework.util.StringUtils.getFilenameExtension(fullName);
    }

    /**
     * 获取文件后缀名 .jpg
     *
     * @param fullName 文件全名
     * @return {String}
     */
    @Nullable
    public static String getFileExtensionWithDot(@Nullable String fullName) {
        if (fullName == null) {
            return null;
        }

        int extIndex = fullName.lastIndexOf(CharPool.DOT);
        if (extIndex == -1) {
            return null;
        }

        int folderIndex = fullName.lastIndexOf(CharPool.SLASH);
        if (folderIndex > extIndex) {
            return null;
        }

        return fullName.substring(extIndex);
    }

    /**
     * 获取文件名，去除后缀名
     *
     * @param path 文件
     * @return {String}
     */
    @Nullable
    public static String getPathWithoutExtension(@Nullable String path) {
        if (path == null) {
            return null;
        }
        return StringUtils.stripFilenameExtension(path);
    }

    /**
     * Returns the path to the system temporary directory.
     *
     * @return the path to the system temporary directory.
     */
    public static String getTempDirPath() {
        return System.getProperty("java.io.tmpdir");
    }

    /**
     * 拼接临时文件目录.
     *
     * @return 临时文件目录.
     */
    public static String toTempDirPath(String subDirFile) {
        return FileUtils.toTempDir(subDirFile).getAbsolutePath();
    }

    /**
     * Returns a {@link File} representing the system temporary directory.
     *
     * @return the system temporary directory.
     */
    public static File getTempDir() {
        return new File(getTempDirPath());
    }

    /**
     * 拼接临时文件目录.
     *
     * @return the system temporary directory.
     */
    public static File toTempDir(String subDirFile) {
        String tempDirPath = FileUtils.getTempDirPath();
        if (subDirFile.startsWith(StringPool.SLASH)) {
            subDirFile = subDirFile.substring(1);
        }
        String fullPath = tempDirPath.concat(subDirFile);
        File fullFilePath = new File(fullPath);
        File dir = fullFilePath.getParentFile();
        if (!dir.exists()) {
            dir.mkdirs();
        }
        return fullFilePath;
    }

    /**
     * Reads the contents of a file into a String.
     * The file is always closed.
     *
     * @param file the file to read, must not be {@code null}
     * @return the file contents, never {@code null}
     */
    public static String readToString(final File file) {
        return readToString(file, Charsets.UTF_8);
    }

    /**
     * Reads the contents of a file into a String.
     * The file is always closed.
     *
     * @param file the file to read, must not be {@code null}
     * @param encoding the encoding to use, {@code null} means platform default
     * @return the file contents, never {@code null}
     */
    public static String readToString(final File file, final Charset encoding) {
        try (InputStream in = Files.newInputStream(file.toPath())) {
            return IoUtils.readToString(in, encoding);
        } catch (IOException e) {
            throw Exceptions.unchecked(e);
        }
    }

    /**
     * Reads the contents of a file into a String.
     * The file is always closed.
     *
     * @param file the file to read, must not be {@code null}
     * @return the file contents, never {@code null}
     */
    public static byte[] readToByteArray(final File file) {
        try (InputStream in = Files.newInputStream(file.toPath())) {
            return IoUtils.readToByteArray(in);
        } catch (IOException e) {
            throw Exceptions.unchecked(e);
        }
    }

    /**
     * Writes a String to a file creating the file if it does not exist.
     *
     * @param file the file to write
     * @param data the content to write to the file
     */
    public static void writeToFile(final File file, final String data) {
        writeToFile(file, data, Charsets.UTF_8, false);
    }

    /**
     * Writes a String to a file creating the file if it does not exist.
     *
     * @param file the file to write
     * @param data the content to write to the file
     * @param append if {@code true}, then the String will be added to the
     *            end of the file rather than overwriting
     */
    public static void writeToFile(final File file, final String data, final boolean append) {
        writeToFile(file, data, Charsets.UTF_8, append);
    }

    /**
     * Writes a String to a file creating the file if it does not exist.
     *
     * @param file the file to write
     * @param data the content to write to the file
     * @param encoding the encoding to use, {@code null} means platform default
     */
    public static void writeToFile(final File file, final String data, final Charset encoding) {
        writeToFile(file, data, encoding, false);
    }

    /**
     * Writes a String to a file creating the file if it does not exist.
     *
     * @param file the file to write
     * @param data the content to write to the file
     * @param encoding the encoding to use, {@code null} means platform default
     * @param append if {@code true}, then the String will be added to the
     *            end of the file rather than overwriting
     */
    public static void writeToFile(final File file, final String data, final Charset encoding, final boolean append) {
        try (OutputStream out = new FileOutputStream(file, append)) {
            IoUtils.write(data, out, encoding);
        } catch (IOException e) {
            throw Exceptions.unchecked(e);
        }
    }

    /**
     * 转成file
     *
     * @param multipartFile MultipartFile
     * @param file File
     */
    public static void toFile(MultipartFile multipartFile, final File file) {
        try {
            FileUtils.toFile(multipartFile.getInputStream(), file);
        } catch (IOException e) {
            throw Exceptions.unchecked(e);
        }
    }

    /**
     * 转成file
     *
     * @param in InputStream
     * @param file File
     */
    public static void toFile(InputStream in, final File file) {
        try (OutputStream out = new FileOutputStream(file)) {
            FileUtils.copy(in, out);
        } catch (IOException e) {
            throw Exceptions.unchecked(e);
        }
    }

    /**
     * Moves a file.
     * <p>
     * When the destination file is on another file system, do a "copy and delete".
     *
     * @param srcFile the file to be moved
     * @param destFile the destination file
     * @throws NullPointerException if source or destination is {@code null}
     * @throws IOException if source or destination is invalid
     * @throws IOException if an IO error occurs moving the file
     */
    public static void moveFile(final File srcFile, final File destFile) throws IOException {
        Assert.notNull(srcFile, "Source must not be null");
        Assert.notNull(destFile, "Destination must not be null");
        if (!srcFile.exists()) {
            throw new FileNotFoundException("Source '" + srcFile + "' does not exist");
        }
        if (srcFile.isDirectory()) {
            throw new IOException("Source '" + srcFile + "' is a directory");
        }
        if (destFile.exists()) {
            throw new IOException("Destination '" + destFile + "' already exists");
        }
        if (destFile.isDirectory()) {
            throw new IOException("Destination '" + destFile + "' is a directory");
        }
        final boolean rename = srcFile.renameTo(destFile);
        if (!rename) {
            FileUtils.copy(srcFile, destFile);
            if (!srcFile.delete()) {
                FileUtils.deleteQuietly(destFile);
                throw new IOException(
                        "Failed to delete original file '" + srcFile + "' after copy to '" + destFile + "'");
            }
        }
    }

    /**
     * Deletes a file, never throwing an exception. If file is a directory, delete it and all
     * sub-directories.
     * <p>
     * The difference between File.delete() and this method are:
     * <ul>
     * <li>A directory to be deleted does not have to be empty.</li>
     * <li>No exceptions are thrown when a file or directory cannot be deleted.</li>
     * </ul>
     *
     * @param file file or directory to delete, can be {@code null}
     * @return {@code true} if the file or directory was deleted, otherwise
     *         {@code false}
     */
    public static boolean deleteQuietly(@Nullable final File file) {
        if (file == null) {
            return false;
        }
        try {
            if (file.isDirectory()) {
                FileSystemUtils.deleteRecursively(file);
            }
        } catch (final Exception ignored) {}

        try {
            return file.delete();
        } catch (final Exception ignored) {
            return false;
        }
    }

    /**
     * NIO 按行读取文件
     *
     * @param path 文件路径
     * @return 行列表
     */
    public static List<String> readLines(String path) {
        return readLines(Paths.get(path));
    }

    /**
     * NIO 按行读取文件
     *
     * @param file 文件
     * @return 行列表
     */
    public static List<String> readLines(File file) {
        return readLines(file.toPath());
    }

    /**
     * NIO 按行读取文件
     *
     * @param path 文件路径
     * @return 行列表
     */
    public static List<String> readLines(Path path) {
        return readLines(path, Charsets.UTF_8);
    }

    /**
     * NIO 按行读取文件
     *
     * @param path 文件路径
     * @param cs 字符集
     * @return 行列表
     */
    public static List<String> readLines(String path, Charset cs) {
        return readLines(Paths.get(path), cs);
    }

    /**
     * NIO 按行读取文件
     *
     * @param file 文件
     * @param cs 字符集
     * @return 行列表
     */
    public static List<String> readLines(File file, Charset cs) {
        return readLines(file.toPath(), cs);
    }

    /**
     * NIO 按行读取文件
     *
     * @param path 文件路径
     * @param cs 字符集
     * @return 行列表
     */
    public static List<String> readLines(Path path, Charset cs) {
        try {
            return Files.readAllLines(path, cs);
        } catch (IOException e) {
            throw Exceptions.unchecked(e);
        }
    }

}
